package cn.itrip.auth.controller;

import cn.itrip.auth.service.TokenService;
import cn.itrip.auth.service.UserService;
import cn.itrip.beans.dto.Dto;
import cn.itrip.beans.pojo.ItripUser;
import cn.itrip.beans.vo.ItripTokenVO;
import cn.itrip.common.DtoUtil;
import cn.itrip.common.EmptyUtils;
import cn.itrip.common.ErrorCode;
import cn.itrip.common.MD5;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.Calendar;

@Controller
@RequestMapping(value = "/api")
public class LoginController {

    @Resource
    private UserService userService;

    @Resource
    private TokenService tokenService;

    @RequestMapping(value="/dologin",method=RequestMethod.POST,produces= "application/json")
    @ResponseBody
    public Dto dologin(@RequestParam String name, @RequestParam String password, HttpServletRequest request) {

        try {
            ItripUser user = userService.login(name, MD5.getMd5(password,32));
            if (EmptyUtils.isNotEmpty(user)){
                String agent = request.getHeader("user-agent");
                String token = tokenService.generateToken(agent,user);
                tokenService.save(token,user);
                //返回ItripTokenVO
                ItripTokenVO tokenVO=new ItripTokenVO(token,
                        Calendar.getInstance().getTimeInMillis()+TokenService.SESSION_TIMEOUT*1000,//2h有效期
                        Calendar.getInstance().getTimeInMillis());

                return DtoUtil.returnDataSuccess(tokenVO);
            } else {
                return DtoUtil.returnFail("用户名密码错误", ErrorCode.AUTH_AUTHENTICATION_FAILED);
            }
        } catch (Exception e) {
            e.printStackTrace();
            return DtoUtil.returnFail(e.getMessage(),ErrorCode.AUTH_AUTHENTICATION_FAILED);
        }
    }

    @RequestMapping(value="/logout",method=RequestMethod.GET,produces= "application/json",headers = "token")
    @ResponseBody
    public Dto logout(HttpServletRequest request) {
        String token = request.getHeader("token");
        String agent = request.getHeader("user-agent");

        try {
            if (tokenService.validate(agent,token)){
                tokenService.delete(token);
                return DtoUtil.returnSuccess();
            } else {
                return DtoUtil.returnFail("token无效",ErrorCode.AUTH_TOKEN_INVALID);
            }
        } catch (Exception e) {
            e.printStackTrace();
            return DtoUtil.returnFail("退出失败",ErrorCode.AUTH_TOKEN_INVALID);
        }
    }
}
